package com.zjsru.oneDay202211;

import java.util.Arrays;
import java.util.Comparator;

/**
 * @Author: CookLee
 * @Date: 2022/11/15
 *
 * 卡车上的最大单元数
 * 二维数组 boxTypes ，其中 boxTypes[i] = [numberOfBoxesi, numberOfUnitsPerBoxi] ：
 * numberOfBoxesi 是类型 i 的箱子的数量。
 * numberOfUnitsPerBoxi 是类型 i 每个箱子可以装载的单元数量。
 * 整数 truckSize 表示卡车上可以装载 箱子 的 最大数量 。只要箱子数量不超过 truckSize ，你就可以选择任意箱子装到卡车上
 *
 *
 * 输入：boxTypes = [[1,3],[2,2],[3,1]], truckSize = 4
 * 输出：8
 * 解释：箱子的情况如下：
 * - 1 个第一类的箱子，里面含 3 个单元。
 * - 2 个第二类的箱子，每个里面含 2 个单元。
 * - 3 个第三类的箱子，每个里面含 1 个单元。
 * 可以选择第一类和第二类的所有箱子，以及第三类的一个箱子。
 * 单元总数 = (1 * 3) + (2 * 2) + (1 * 1) = 8
 *
 */
public class MaximumUnits {
    
    /**
     * 先排序，再计算
     */
    public int maximumUnits(int[][] boxTypes, int truckSize) {
        int result = 0;
        Arrays.sort(boxTypes, Comparator.comparingInt(o -> o[1]));
        for (int i = boxTypes.length - 1; i >= 0; i--) {
            if (truckSize > boxTypes[i][0]) {
                //首先添加 最大价值的盒子总值
                //boxTypes[i][0]即是数量，boxTypes[i][1]可装的单元数量
                result += boxTypes[i][0] * boxTypes[i][1];
                truckSize = truckSize - boxTypes[i][0];
            } else {
                //获取最后 可装载盒子的总值
                result += truckSize * boxTypes[i][1];
                return result;
            }
        }
        return result;
    }
    
    /**
     * 贪心
     */
    public int maximumUnits2(int[][] boxTypes, int truckSize) {
        int n = boxTypes.length, ans = 0;
        Arrays.sort(boxTypes, (a, b) -> b[1] - a[1]);
        for (int i = 0, cnt = 0; i < n && cnt < truckSize; i++) {
            int a = boxTypes[i][0], b = boxTypes[i][1], c = Math.min(a, truckSize - cnt);
            cnt += c;
            ans += c * b;
        }
        return ans;
    }
    
    
    public static void main(String[] args) {
        MaximumUnits maximumUnits = new MaximumUnits();
        int[][] boxTypes = new int[][] {{1, 3}, {2, 2}, {3, 1}};
        int truckSize = 4;
        System.out.println(maximumUnits.maximumUnits(boxTypes, truckSize));
    }
}
